# 第57题

# 问题：定义一个自定义异常类，它将字符串消息作为属性。

# 提示：要定义一个自定义异常，我们需要定义一个继承自exception的类。
class NotStringError(Exception):
    def __init__(self, error_msg) -> None:
        self.error_msg = error_msg

    def __str__(self) -> str:
        return self.error_msg

if __name__ == '__main__':
    my_error = NotStringError('只允许输入字符串')
    my_str = 1234
    if type(my_str) == str:   # 类型比较  
        print(my_str)
    else:
        raise my_error